package com.haidechizi.leetcode;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.Stack;

/**
 * 给定一个数组 candidates 和一个目标数 target ，找出 candidates 中所有可以使数字和为 target 的组合。
 * <p>
 * candidates 中的每个数字在每个组合中只能使用一次。
 * <p>
 * 说明：
 * <p>
 * 所有数字（包括目标数）都是正整数。
 * 解集不能包含重复的组合。 
 * 示例 1:
 * <p>
 * 输入: candidates = [10,1,2,7,6,1,5], target = 8,
 * 所求解集为:
 * [
 * [1, 7],
 * [1, 2, 5],
 * [2, 6],
 * [1, 1, 6]
 * ]
 * 示例 2:
 * <p>
 * 输入: candidates = [2,5,2,1,2], target = 5,
 * 所求解集为:
 * [
 *   [1,2,2],
 *   [5]
 * ]
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/combination-sum-ii
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class _40_组合总和II {
    public List<List<Integer>> combinationSum2(int[] candidates, int target) {
        List<List<Integer>> result = new ArrayList<>();
        Stack<Integer> stack = new Stack<>();
        Set<List<Integer>> set = new HashSet<>();
        Arrays.sort(candidates);
        generateSum(0, candidates, stack, set, target);
        result.addAll(set);
        return result;

    }

    private void generateSum(int i, int[] candidates, Stack<Integer> stack, Set<List<Integer>> set, int target) {
        if (i >= candidates.length) {
            return;
        }
        stack.push(candidates[i]);
        boolean flag = add(set, stack, target);
        if(flag) {
            generateSum(i + 1, candidates, stack, set, target);
        }

        stack.pop();
        generateSum(i + 1, candidates, stack, set, target
        );
    }

    private boolean add(Set<List<Integer>> set, Stack<Integer> stack, int target) {
        int sum = 0;
        for (int i = 0; i < stack.size(); i++) {
            sum += stack.get(i);
        }
        if (sum == target) {
            List<Integer> list = new ArrayList<>();
            list.addAll(stack);
            set.add(list);
            return false;

        }
        if(sum > target) {
            return false;
        }
        return true;
    }
}
